# -*- mode: python ; coding: utf-8 -*-
"""
角标识别程序 PyInstaller 配置文件
优化版本 - 解决 PaddlePaddle 和 Cython 打包问题
"""

import os
import sys
from PyInstaller.utils.hooks import collect_data_files, collect_submodules, collect_dynamic_libs

block_cipher = None

# ==================== 数据文件收集 ====================
datas = []

# 收集 PaddleOCR 数据文件
try:
    datas += collect_data_files('paddleocr', include_py_files=True)
    print("✓ 已收集 paddleocr 数据文件")
except:
    print("⚠ 未找到 paddleocr 数据文件")

# 收集 Paddle 数据文件
try:
    datas += collect_data_files('paddle', include_py_files=True)
    print("✓ 已收集 paddle 数据文件")
except:
    print("⚠ 未找到 paddle 数据文件")

# 收集 Cython 工具文件（关键！）
try:
    import Cython
    cython_dir = os.path.dirname(Cython.__file__)
    utility_dir = os.path.join(cython_dir, 'Utility')
    includes_dir = os.path.join(cython_dir, 'Includes')
    compiler_dir = os.path.join(cython_dir, 'Compiler')
    
    if os.path.exists(utility_dir):
        datas.append((utility_dir, 'Cython/Utility'))
        print("✓ 已收集 Cython/Utility")
    
    if os.path.exists(includes_dir):
        datas.append((includes_dir, 'Cython/Includes'))
        print("✓ 已收集 Cython/Includes")
    
    if os.path.exists(compiler_dir):
        # 收集 Compiler 目录下的所有文件
        datas.append((compiler_dir, 'Cython/Compiler'))
        print("✓ 已收集 Cython/Compiler")
except ImportError:
    print("⚠ 未找到 Cython")

# 收集 PaddleOCR 模型目录（关键！必须存在）
if os.path.exists('paddleocr'):
    # 收集整个 paddleocr 目录及其子目录
    for root, dirs, files in os.walk('paddleocr'):
        for file in files:
            file_path = os.path.join(root, file)
            # 计算相对路径
            rel_path = os.path.relpath(root, '.')
            datas.append((file_path, rel_path))
    print("✓ 已收集本地 paddleocr 模型目录")
else:
    print("⚠️ 警告: paddleocr 模型目录不存在！")
    print("   请先运行: prepare_models.bat")

# ==================== 二进制文件收集 ====================
binaries = []

# 收集 paddle 的动态库
try:
    binaries += collect_dynamic_libs('paddle')
    print("✓ 已收集 paddle 动态库")
except:
    print("⚠ 未找到 paddle 动态库")

# 收集 cv2 的动态库
try:
    binaries += collect_dynamic_libs('cv2')
    print("✓ 已收集 cv2 动态库")
except:
    print("⚠ 未找到 cv2 动态库")

# ==================== 隐藏导入 ====================
hiddenimports = []

# PaddleOCR 相关
hiddenimports += [
    'paddleocr',
    'paddleocr.paddleocr',
    'paddleocr.tools.infer',
    'paddle',
    'paddle.base',
    'paddle.base.framework',
    'paddle.base.dataset',
    'paddle.utils',
    'paddle.utils.cpp_extension',
    'paddle.utils.cpp_extension.cpp_extension',
]

# Cython 相关（关键！）
hiddenimports += [
    'Cython',
    'Cython.Compiler',
    'Cython.Compiler.Main',
    'Cython.Compiler.Symtab',
    'Cython.Compiler.PyrexTypes',
    'Cython.Compiler.Code',
    'Cython.Utils',
]

# Setuptools 相关
hiddenimports += [
    'setuptools',
    'setuptools.command',
    'setuptools.command.build_ext',
]

# OpenCV 相关
hiddenimports += [
    'cv2',
    'numpy',
]

# 其他依赖
hiddenimports += [
    'PIL',
    'PIL.Image',
    'openpyxl',
    'qrcode',
    'svgwrite',
]

# 自动收集子模块
try:
    hiddenimports += collect_submodules('paddleocr')
    hiddenimports += collect_submodules('paddle')
    print("✓ 已自动收集 paddle/paddleocr 子模块")
except:
    print("⚠ 自动收集子模块失败")

# ==================== 分析 ====================
a = Analysis(
    ['recognition.py'],
    pathex=[],
    binaries=binaries,
    datas=datas,
    hiddenimports=hiddenimports,
    hookspath=[],
    hooksconfig={},
    runtime_hooks=[],
    excludes=[
        'matplotlib',  # 如果不需要可以排除
        'tkinter',     # 如果不需要可以排除
    ],
    win_no_prefer_redirects=False,
    win_private_assemblies=False,
    cipher=block_cipher,
    noarchive=False,
)

# ==================== 打包配置 ====================
pyz = PYZ(a.pure, a.zipped_data, cipher=block_cipher)

# 使用文件夹模式（更稳定）
exe = EXE(
    pyz,
    a.scripts,
    [],
    exclude_binaries=True,
    name='角标识别',
    debug=False,
    bootloader_ignore_signals=False,
    strip=False,
    upx=True,
    console=True,  # 显示控制台窗口
    disable_windowed_traceback=False,
    argv_emulation=False,
    target_arch=None,
    codesign_identity=None,
    entitlements_file=None,
    icon=None,
)
coll = COLLECT(
    exe,
    a.binaries,
    a.zipfiles,
    a.datas,
    strip=False,
    upx=True,
    upx_exclude=[],
    name='角标识别',
)


